#!/usr/bin/env python
# -*- encoding: utf-8 -*-
"""
@File    :   train.py.py    
@Contact :   ruaqy@qq.com
@License :   (C)Copyright 2022-~, GPL 2.0

@Modify Time      @Author    @Version    @Description
------------      -------    --------    -----------
2022/5/16 21:48   rqy        1.0         None
"""

from model.utils.args import parser_init

if __name__ == '__main__':
    # 初始化解析器
    parser = parser_init()

    # 解析参数
    args, unknown_args = parser.parse_known_args()
    if args.target == 'mindspore':
        import run.runin_mindspore
        run.runin_mindspore.main(args)
    elif args.target == 'pytorch':
        import run.runin_pytorch
        run.runin_pytorch.main(args)
